import mongoose from 'mongoose';
import dotenv from 'dotenv';
import User from '../models/User';

dotenv.config();

const initializeDatabase = async () => {
  try {
    // 连接数据库
    await mongoose.connect(process.env.MONGODB_URI || 'mongodb://localhost:27017/tourism_management');
    console.log('MongoDB 连接成功');

    // 清空现有数据
    await User.deleteMany({});
    console.log('清空现有用户数据');

    // 创建管理员用户（插入明文密码）
    const admin = await User.create({
      username: 'admin',
      email: 'admin@example.com',
      password: 'admin123', // 明文
      role: 'admin',
      avatar: 'https://api.dicebear.com/7.x/avataaars/svg?seed=admin'
    });
    console.log('创建管理员用户成功:', admin.username);

    // 创建测试用户（插入明文密码）
    const testUser = await User.create({
      username: 'test',
      email: 'test@example.com',
      password: 'test123', // 明文
      role: 'user',
      avatar: 'https://api.dicebear.com/7.x/avataaars/svg?seed=test'
    });
    console.log('创建测试用户成功:', testUser.username);

    console.log('数据库初始化完成');
    process.exit(0);
  } catch (error) {
    console.error('数据库初始化失败:', error);
    process.exit(1);
  }
};

initializeDatabase(); 